#pragma once

#include <iostream>
#include <string>
#include <pthread.h>

typedef void (*func_t)(void*);

class Thread
{
public:
    typedef enum
    {
        NEW = 0,
        RUNNING,
        EXITED
    } ThreadStatus;

    // Thread()
    // { }

    Thread(int num, func_t func, void* args):_tid(0), _status(NEW), _func(func), _args(args)
    {
        char name[128];
        int n = sizeof(name) / sizeof(char);
        snprintf(name, n, "thread-%d", num);
        _name = name;
    }

    int status() { return _status; }
    std::string name() { return _name; }
    pthread_t threadid()
    {
        if (_status == RUNNING)
        {
            return _tid;
        }
        else
        {
            return 0;
        }
    }

    void operator()()
    {
        if (_func != nullptr) _func(_args);
    }

    static void* runHelper(void* args)
    {
        Thread* ts = (Thread*)args;
        (*ts)();
        return nullptr;
    }

    void run()
    {
        int n = pthread_create(&_tid, nullptr, runHelper, this);
        if (n != 0) exit(1);
        _status = RUNNING;
    }

    void join()
    {
        int n = pthread_join(_tid, nullptr);
        if (n != 0)
        {
            std::cerr << "main thread join thread" << _name << "error" << std::endl;
            return;
        }
        _status = EXITED;
    }

    ~Thread()
    {}

private:
    pthread_t _tid;
    std::string _name;
    func_t _func;
    void* _args;
    ThreadStatus _status;
};